import unittest
from main import extract_group_from_fn

class MyTestCase(unittest.TestCase):
    def test_something(self):
        self.assertEqual('00J000524_义务教育', extract_group_from_fn('00J000524_义务教育_1943年第1卷第1期.pdf'))  # add assertion here

    def test_something(self):
        self.assertEqual('1_w', extract_group_from_fn('1_w_x_2.pdf'))  # add assertion here

if __name__ == '__main__':
    unittest.main()
